package com.myjpa.generate;

import org.junit.Test;

public class TaskGenerator {

	boolean override = false;

	private GenerateContext initGenerateContext() {
		GenerateContext ctx = new GenerateContext();
		ctx.setRootPath("/Users/devsaga/workspace/idea/myjpa/myjpa-querydsl-5");
		ctx.setResponsePackage("com.myjpa.common.web");
		ctx.setCriteriaPackage("com.myjpa.common.criteria");
		ctx.setPkg("com.jpa");
		ctx.setOverrideFile(override);
		return ctx;
	}

	@Test
	public void initTask() {
		Generator generator = new Generator();
		GenerateContext ctx = initGenerateContext();
		ctx.setClassName("Task");
		ctx.setTableName("t_task");
		ctx.setMappingPath("/task");
		ctx.setOverrideFile(true);
		ctx.addAttribute("name", String.class);
		ctx.addAttribute("note", String.class);
		ctx.addAttribute("groupId", String.class);
		generator.generate(ctx);
	}


	@Test
	public void initClassDef() {
		Generator generator = new Generator();
		GenerateContext ctx = initGenerateContext();
		// 配置主键及生成策略
		ctx.setPrimaryKey("id", "String", GenerateContext.PrimaryKey.GENERATION_TYPE_ASSIGNED);
		ctx.setClassName("ClassDef");
		ctx.setTableName("rwd_def_class");
		ctx.setMappingPath("/class");
		ctx.setOverrideFile(true);
		ctx.setWithModel(true); // 生成model类

		// 从数据库解析字段
		String url = "";
		String username = "";
		String password = "";
		String tableSchema = "";
		JdbcGeneratorUtils.initAttribute(url, username, password, tableSchema, ctx);

		generator.generate(ctx);
	}

}
